package com.heima.wemedia.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollStreamUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.stream.StreamUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSON;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.heima.common.constants.WemediaConstants;
import com.heima.common.dtos.PageResponseResult;
import com.heima.common.dtos.ResponseResult;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.exception.CustomException;
import com.heima.model.wemedia.dtos.WmMaterialDto;
import com.heima.model.wemedia.dtos.WmNewsDto;
import com.heima.model.wemedia.dtos.WmNewsPageReqDto;
import com.heima.model.wemedia.pojos.WmMaterial;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.model.wemedia.pojos.WmNewsMaterial;
import com.heima.wemedia.mapper.WmMaterialMapper;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.mapper.WmNewsMaterialMapper;
import com.heima.wemedia.service.WmMaterialService;
import com.heima.wemedia.service.WmNewsAutoScanService;
import com.heima.wemedia.service.WmNewsService;
import com.heima.wemedia.utils.WmThreadLocalUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.sql.Wrapper;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

@Service
@Slf4j
@Transactional
public class WmNewsServiceImpl extends ServiceImpl<WmNewsMapper, WmNews> implements WmNewsService {

    @Resource
    private WmMaterialService wmMaterialService;


    @Resource
    private WmMaterialMapper wmMaterialMapper;

    @Resource
    private WmNewsMaterialMapper wmNewsMaterialMapper;

    @Resource
    private WmNewsAutoScanService wmNewsAutoScanService;


    @Override
    public ResponseResult<List<WmNews>> findAll(WmNewsPageReqDto dto) {
//        查询啊 ，参数有点多啊。
//        1，检查参数
        if (ObjectUtil.isEmpty(dto)) {
            return ResponseResult.errorResult(AppHttpCodeEnum.PARAM_INVALID);
        }
//        检查分页参数
        dto.checkParam();
//        获取当前登陆人信息.
        Integer id = WmThreadLocalUtil.get().getId();
//        然后分页参数
        IPage<WmNews> page = new Page<>(dto.getPage(), dto.getSize());
//        条件.
        LambdaQueryWrapper<WmNews> wrapper1 = Wrappers.lambdaQuery();

        wrapper1
                .eq(ObjectUtil.isNotEmpty(dto.getChannelId()), WmNews::getChannelId, dto.getChannelId())
                .eq(WmNews::getUserId, id)
                .eq(ObjectUtil.isNotEmpty(dto.getStatus()), WmNews::getStatus, dto.getStatus())
                .like(ObjectUtil.isNotEmpty(dto.getKeyword()), WmNews::getTitle, dto.getKeyword())
                .between(ObjectUtil.isAllNotEmpty(dto.getBeginPubDate(), dto.getEndPubDate()), WmNews::getPublishTime, dto.getBeginPubDate(), dto.getEndPubDate())  //时间范围查询
                .orderByDesc(WmNews::getCreatedTime);
        IPage<WmNews> page1 = this.page(page, wrapper1);
        //3.结果返回
        return PageResponseResult.<List<WmNews>>builder()
                .size(dto.getSize())
                .currentPage(dto.getPage())
                .total(dto.getPage())
                .size(Convert.toInt(page.getTotal()))
                .data(page.getRecords())
                .build();
    }

    /**
     * 改进后的代码。
     *
     * 1. 前端提交发布或保存为草稿
     * 2. 后台判断请求中是否包含了文章id
     * 3. 如果不包含id,则为新增
     *   3.1. 执行新增文章的操作
     *   3.2. 关联文章内容图片与素材的关系
     *   3.3. 关联文章封面图片与素材的关系
     * 4. 如果包含了id，则为修改请求
     *   4.1. 删除该文章与素材的所有关系
     *   4.2. 执行修改操作
     *   4.3. 关联文章内容图片与素材的关系
     *   4.4. 关联文章封面图片与素材的关系
     * @param dto
     * @return
     */
    @Override
    public ResponseResult submitNews(WmNewsDto dto) {
//        0.条件判断。
        if(ObjectUtil.isEmpty(dto.getContent())){
            return ResponseResult.errorResult(AppHttpCodeEnum.PARAM_INVALID);
        }
//       1,保存或修改文章。 dto  转换实体类
        WmNews wmNews = BeanUtil.toBean(dto, WmNews.class);

//        封面图片  list----> string
        if(CollUtil.isNotEmpty(dto.getImages())){
            String imageStr = StrUtil.join(",", dto.getImages());
            wmNews.setImages(imageStr);
        }
//       判断如果当前封面类型为 -1 表示自动类型  这个地方因为数据库限制原因。
        if(ObjectUtil.equal(dto.getType(),WemediaConstants.WM_NEWS_TYPE_AUTO)){
            wmNews.setType(null);
        }
//        上面都是对前端传来的参数 进行校验。

//        保存或者更新数据.
        this.saveOrUpdateWmNews(wmNews);

//        判断是否是草稿，如果是草稿，就结束当前方法。
        if(ObjectUtil.equal(dto.getStatus(),WmNews.Status.NORMAL.getCode())){
            return ResponseResult.okResult(AppHttpCodeEnum.SUCCESS);
        }

//        3,如果部署草稿，保存文章内容图片与素材的关系。
//        获取到文章内容中的图片信息。
        List<String> materials = extractUrlInfo(dto.getContent());

//        然后保存 newsId 和 素材 id。  id 自动回填机制。
        saveRelativeInfoForContent(materials,wmNews.getId());
        //4.不是草稿，保存文章封面图片与素材的关系，如果当前布局是自动，需要匹配封面图片
        saveRelativeInfoForCover(dto, wmNews, materials);

//        这个时候进行自动审核。
        this.wmNewsAutoScanService.autoScanWmNews(wmNews.getId());
        return ResponseResult.okResult(AppHttpCodeEnum.SUCCESS);

    }


    /**
     * 处理文章内容图片与素材的关系
     * 文章内容与素材之间的关系
     * 封面与素材之间的关系。
     */
    private void saveRelativeInfoForContent(List<String> materials, Integer newsId) {
        saveRelativeInfo(materials, newsId, WemediaConstants.WM_CONTENT_REFERENCE);
    }


    /**
     * 保存文章图片与素材的关系到数据库中
     */
    private void saveRelativeInfo(List<String> materials, Integer newsId, Short type){
//        判断是否为null
        if(CollUtil.isEmpty(materials)){
            return ;
        }
//      通过 图片的url 查询素材的id。
        List<WmMaterial> dbMaterials = wmMaterialMapper.selectList(Wrappers.<WmMaterial>lambdaQuery().in(WmMaterial::getUrl, materials));
//        判断素材是否有效。
        if(CollUtil.isEmpty(dbMaterials)){
//            手动抛出异常，
            throw new CustomException(AppHttpCodeEnum.SERVER_ERROR);
        }

        List<Integer> idList = CollStreamUtil.toList(dbMaterials, WmMaterial::getId);
//        批量保存。
        this.wmNewsMaterialMapper.saveRelations(idList,newsId,type);



    }


    /**
     * 第一个功能：如果当前封面类型为自动，则设置封面类型的数据
     * 匹配规则：
     * 1，如果内容图片大于等于1，小于3  单图  type 1
     * 2，如果内容图片大于等于3  多图  type 3
     * 3，如果内容没有图片，无图  type 0
     * <p>
     * 第二个功能：保存封面图片与素材的关系
     */
    private void saveRelativeInfoForCover(WmNewsDto dto, WmNews wmNews, List<String> materials) {
        List<String> images = dto.getImages();
        //如果当前封面类型为自动，则设置封面类型的数据
        if (ObjectUtil.equal(dto.getType(), WemediaConstants.WM_NEWS_TYPE_AUTO)) {
            if (CollUtil.isEmpty(materials)) {
                //无图
                wmNews.setType(WemediaConstants.WM_NEWS_NONE_IMAGE);
            } else if (CollUtil.size(materials) >= 3) {
                //多图
                wmNews.setType(WemediaConstants.WM_NEWS_MANY_IMAGE);
                images = CollUtil.sub(materials, 0, 3);
            } else {
                //单图
                wmNews.setType(WemediaConstants.WM_NEWS_SINGLE_IMAGE);
                images = CollUtil.sub(materials, 0, 1);
            }
            //修改文章
            if (CollUtil.isNotEmpty(images)) {
                wmNews.setImages(StrUtil.join(",", images));
            }
            super.updateById(wmNews);
        }
        if (CollUtil.isNotEmpty(images)) {
            saveRelativeInfo(images, wmNews.getId(), WemediaConstants.WM_COVER_REFERENCE);
        }
    }

    /**
     * 把文章内容用集合返回
     * @param content  String 到集合返回
     * @return
     */
    private List<String> extractUrlInfo(String content) {
        return StreamUtil.of(JSONUtil.toList(content,JSONObject.class))
                .filter(jsonObject -> StrUtil.equals(jsonObject.getStr("type"),"image"))
                .map(jsonObject -> jsonObject.getStr("value"))
                .collect(Collectors.toList());

    }

    /**
     * 根据 id 来选择 插入还是更新，因为 前端传来的dto 数据不全，需要参数进行补充。
     * @param wmNews
     */
    private void saveOrUpdateWmNews(WmNews wmNews) {
        if(ObjectUtil.isEmpty(wmNews.getId())){
//            补全属性。
            wmNews.setUserId(WmThreadLocalUtil.get().getId());
            wmNews.setCreatedTime(new Date());
            wmNews.setSubmitedTime(wmNews.getCreatedTime());
            wmNews.setEnable((short)1);
//            保存
            super.save(wmNews);
            return;
        }
//        id 存在 做更新操作。首先，先删除图片和素材的关系。
        LambdaQueryWrapper<WmNewsMaterial> queryWrapper = Wrappers.<WmNewsMaterial>lambdaQuery().eq(WmNewsMaterial::getNewsId, wmNews.getId());
        this.wmNewsMaterialMapper.delete(queryWrapper);
//        修改数据
        super.updateById(wmNews);
    }

    /**
     * 自己梳理流程。
     * @param dto
     * @return
     */
    public ResponseResult submitNews11(WmNewsDto dto) {
        WmNews wmNews = BeanUtil.toBean(dto, WmNews.class);
//      做了一层 判断 图片是否为null,如果为不为null之后。
        if(CollUtil.isNotEmpty(dto.getImages())){
            String join = StrUtil.join(",", dto.getImages());
            wmNews.setImages(join);
        }
//        判断是否存在id.
        if(ObjectUtil.isEmpty(dto.getId())){
//            如果id为null 做插入操作。
//            填充数据。
//            1，那个自媒体对应的用户id？
            wmNews.setUserId(WmThreadLocalUtil.get().getId());
//            2，创建时间和更新时间。
            wmNews.setCreatedTime(new Date());
            wmNews.setSubmitedTime(wmNews.getCreatedTime());
            wmNews.setEnable((short)1);
//            4，然后存放进去即可。
            super.save(wmNews);

        }else{
//            如果id 存在 就是删除该关系表对应的素材信息删掉。引入素材的service。
            // 存在id 做更新，不存在id做插入。
//        然后添加 news  和 materials 的 的关系表，先查素材在哪。
//        修改的前提是查找到素材在表中对应的字段。
            LambdaQueryWrapper<WmNewsMaterial> wrapper = Wrappers.lambdaQuery();
            wrapper.eq(WmNewsMaterial::getNewsId,dto.getId());
            this.wmNewsMaterialMapper.delete(wrapper);
//       然后进行更新 news 表
            super.updateById(wmNews);
        }


        //2.判断是否为草稿  如果为草稿结束当前方法
//        草稿，不做 关系表的 插入。
        if (ObjectUtil.equal(dto.getStatus(), WmNews.Status.NORMAL.getCode())) {
            return ResponseResult.okResult(AppHttpCodeEnum.SUCCESS);
        }


//      拿到content里面的内容。
//      新增完成文章之后，把表之间关联出来，去查询news id对应的素材id。
//      把Content 里面内容 查出来对应的 素材id。
        List<String> maList = StreamUtil.of(JSONUtil.toList(dto.getContent(), JSONObject.class))
                .filter(jsonObject -> StrUtil.equals(jsonObject.getStr("type"), "image"))
                .map(jsonObject -> jsonObject.getStr("value"))
                .collect(Collectors.toList());
//        然后批量查找就行了。
//        List<WmMaterial> wmMaterialList = this.wmMaterialService.lambdaQuery().in(WmMaterial::getUrl, maList).list();
        //通过图片的url查询素材的id
        List<WmMaterial> wmMaterialList = this.wmMaterialMapper.selectList(Wrappers.<WmMaterial>lambdaQuery().in(WmMaterial::getUrl, maList));

//        问题所在 ，如果图片一样查找到图片个数不会重复

//       判断是否查找到了
        if(CollUtil.isEmpty(wmMaterialList)){
            throw  new CustomException(AppHttpCodeEnum.DATA_NOT_EXIST);
        }
////        查找到的数据匹配？问题所在 ，如果图片一样查找到图片个数不会重复
//        if(ObjectUtil.notEqual(wmMaterialList.size(),maList.size())){
//            throw  new CustomException(AppHttpCodeEnum.SERVER_ERROR);
//        }
//        匹配之后，然后进行 map 转 成对应 素材id 集合。
        List<Integer> idList = CollStreamUtil.toList(wmMaterialList, WmMaterial::getId);
//        然后进行批量添加到 关系表中。
        this.wmNewsMaterialMapper.saveRelations(idList, wmNews.getId(),dto.getType());

//        处理完，内容和和素材的关系表了,
//        接下来处理 封面和素材的关系表。
        /**
         * 第一个功能: 如果当前类型为自动，则设置封面类型的数据,
         * 匹配规则：
         * 1,如果内容图片大于等于1， 小于3 单图 type 1.
         * 2,如果内容图片大于等于3， 多图 type 3
         * 3，如果内容没有图片，无图， type 0
         *
         * 然后保存 封面图片和素材的关系。
         */
        if(ObjectUtil.equal(dto.getTitle(), WemediaConstants.WM_NEWS_TYPE_AUTO)){
            List<String> images = dto.getImages();
//            先判断内容里面的图片
            if(CollUtil.isEmpty(maList)){
                //无图, 无图。
                wmNews.setType(WemediaConstants.WM_NEWS_NONE_IMAGE);
            }else if(CollUtil.size(maList) >= 3 ){
                // 多图。
                wmNews.setType(WemediaConstants.WM_NEWS_MANY_IMAGE);
//                进行了截取。
                images = CollUtil.sub(maList,0,3);
            }else{
//                单图
                wmNews.setType(WemediaConstants.WM_NEWS_SINGLE_IMAGE);
//                截取一张作为 封面。
                images = CollUtil.sub(maList,0,1);
            }
//            然后修改文章即可。
            String join = StrUtil.join(",", images);
            if(CollUtil.isNotEmpty(images)){
                wmNews.setImages(join);
            }
//            上面是把封面放进去。  因为已经插入了，有了自动回填的操作。
            super.updateById(wmNews);
//            然后就保存封面和 素材之间的对应关系。

            List<String> maListCover = StreamUtil.of(JSONUtil.toList(join, JSONObject.class))
                    .filter(jsonObject -> StrUtil.equals(jsonObject.getStr("type"), "image"))
                    .map(jsonObject -> jsonObject.getStr("value"))
                    .collect(Collectors.toList());

            List<WmMaterial> wmMaterialList1 = this.wmMaterialService.lambdaQuery().in(WmMaterial::getUrl, maListCover).list();
            //        匹配之后，然后进行 map 转 成对应 素材id 集合。
            List<Integer> idList1 = CollStreamUtil.toList(wmMaterialList, WmMaterial::getId);
//        然后进行批量添加到 关系表中。
            this.wmNewsMaterialMapper.saveRelations(idList1, wmNews.getId(),dto.getType());

        }
        return  ResponseResult.okResult(AppHttpCodeEnum.SUCCESS);
    }



}
